/*
 * Copyright (c) 2010-2015 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <vector>
#include <string>
#include <boost/optional.hpp>
#include <boost/shared_ptr.hpp>

#include <Swiften/Base/API.h>
#include <Swiften/Elements/ToplevelElement.h>

namespace Swift {
	class SWIFTEN_API StreamFeatures : public ToplevelElement	{
		public:
			typedef boost::shared_ptr<StreamFeatures> ref;

			StreamFeatures() : hasStartTLS_(false), hasResourceBind_(false), hasSession_(false), hasStreamManagement_(false), hasRosterVersioning_(false) {}

			void setHasStartTLS() {
				hasStartTLS_ = true;
			}

			bool hasStartTLS() const {
				return hasStartTLS_;
			}

			void setHasSession() {
				hasSession_ = true;
			}

			bool hasSession() const {
				return hasSession_;
			}

			void setHasResourceBind() {
				hasResourceBind_ = true;
			}

			bool hasResourceBind() const {
				return hasResourceBind_;
			}

			const std::vector<std::string>& getCompressionMethods() const {
				return compressionMethods_;
			}

			void addCompressionMethod(const std::string& mechanism) {
				compressionMethods_.push_back(mechanism);
			}

			bool hasCompressionMethod(const std::string& mechanism) const;

			const std::vector<std::string>& getAuthenticationMechanisms() const {
				return authenticationMechanisms_;
			}

			void addAuthenticationMechanism(const std::string& mechanism) {
				authenticationMechanisms_.push_back(mechanism);
			}

			bool hasAuthenticationMechanism(const std::string& mechanism) const;

			bool hasAuthenticationMechanisms() const {
				return !authenticationMechanisms_.empty();
			}

			const boost::optional<std::string> getAuthenticationHostname() const {
				return authenticationHostname_;
			}

			void setAuthenticationHostname(const boost::optional<std::string> authenticationHostname) {
				authenticationHostname_ = authenticationHostname;
			}

			bool hasStreamManagement() const {
				return hasStreamManagement_;
			}

			void setHasStreamManagement() {
				hasStreamManagement_ = true;
			}

			bool hasRosterVersioning() const {
				return hasRosterVersioning_;
			}

			void setHasRosterVersioning() {
				hasRosterVersioning_ = true;
			}

		private:
			bool hasStartTLS_;
			std::vector<std::string> compressionMethods_;
			std::vector<std::string> authenticationMechanisms_;
			bool hasResourceBind_;
			bool hasSession_;
			bool hasStreamManagement_;
			bool hasRosterVersioning_;
			boost::optional<std::string> authenticationHostname_;
	};
}
